////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2008 Michiel Helvensteijn                                    //
//                                                                            //
// This file is part of the Mist compiler.                                    //
//                                                                            //
// The Mist compiler is free software: you can redistribute it and/or modif   //
// it under the terms of the GNU General Public License as published by       //
// the Free Software Foundation, either version 3 of the License, or          //
// (at your option) any later version.                                        //
//                                                                            //
// The Mist compiler is distributed in the hope that it will be useful,       //
// but WITHOUT ANY WARRANTY; without even the implied warranty of             //
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the              //
// GNU General Public License for more details.                               //
//                                                                            //
// You should have received a copy of the GNU General Public License          //
// along with the Mist compiler.  If not, see <http://www.gnu.org/licenses/>. //
////////////////////////////////////////////////////////////////////////////////

#include "visitors/findreferencesvisitor.hpp"

/******************************************************************************
 * Includes                                                                   *
 ******************************************************************************/

#include <boost/shared_ptr.hpp>
using boost::dynamic_pointer_cast;

#include "ast/nodetypes.hpp"
#include "visitors/expressionnavigator.hpp"

/******************************************************************************
 * FindReferencesVisitor Class                                                *
 ******************************************************************************/

class FindReferencesVisitor : public ExpressionNavigator {
	public:
		
		void visit(shared_ptr<Reference> node);
		
		set<shared_ptr<Reference> > references();
	
	private:
		
		set<shared_ptr<Reference> > _references;
};

/******************************************************************************
 * Function Implementations                                                   *
 ******************************************************************************/

set<shared_ptr<Reference> > find_references(shared_ptr<Node> node) {
	assert(node);
	
	shared_ptr<FindReferencesVisitor> visitor(new FindReferencesVisitor);
	node->accept(visitor);
	
	return visitor->references();
}

/******************************************************************************/

void FindReferencesVisitor::visit(shared_ptr<Reference> node) {
	_references.insert(node);
}

set<shared_ptr<Reference> > FindReferencesVisitor::references() {
	return _references;
}
